#include <iostream>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <sys/types.h>
#include <string>
#include "common.hpp"

// 用于创建共享内存
const int gdefaultid = -1;
const int gsize = 4096;

// 指明用户类型
#define CREATER "creater"
#define USER "user"

// 用于构建唯一key值
std::string pathname = ".";
const int proj_id = 0x4444; // 随便给值只要保证构建的key值唯一即可，若是不唯一传不同的pathname和proj_id

class shm
{
private:
    void CreateOrGet(int flag)
    {
        _shmid = shmget(_key, _size, flag);
        if (_shmid == -1)
        {
            ERR_EXIT("shmget");
        }
        printf("shmget success , shmid : %d\n", _shmid);
    }
    // 申请共享内存
    void Create()
    {
        CreateOrGet(IPC_CREAT | IPC_EXCL | 0666);
    }
    // 获取共享内存
    void Get()
    {
        CreateOrGet(IPC_CREAT);
    }

    // 实现进程与共享内存关联
    void Attach()
    {
        _startaddr = shmat(_shmid, NULL, 0);
        if ((long long)_startaddr < 0)
        {
            ERR_EXIT("shmat");
        }
        printf("shmat success\n");
    }
    // 删除共享内存
    void Destroy()
    {
        if (_shmid == -1)
        {
            printf("nothing to delete\n");
            return;
        }
        else
        {
            int n = shmctl(_shmid, IPC_RMID, NULL);
            if (n == -1)
            {
                ERR_EXIT("shmctl");
            }
            printf("shmctl->IPC_RMID success\n");
        }
    }

public:
    shm(const std::string &usertype, const std::string pathname, int proj_id)
        : _shmid(gdefaultid),
          _size(gsize),
          _startaddr(NULL),
          _usertype(usertype)
    {
        // 生成一个key值
        _key = ftok(pathname.c_str(), proj_id);
        if (_key == -1)
        {
            ERR_EXIT("ftok");
        }
        printf("key success : 0x%x\n", _key);
        // key值构建成功，开始申请共享内存

        // 在构造函数里面申请或者获取共享内存并且关联共享内存和进程
        if (_usertype == CREATER)
        {
            Create();
        }
        else if (_usertype == USER)
        {
            Get();
        }
        else
        {
        }

        Attach();
    }

    // 打印虚拟起始地址
    void *VirtualAddr()
    {
        printf("VirtualAddr : %p\n", _startaddr);
        return _startaddr;
    }
    int Size()
    {
        printf("size : %d\n", _size);
        return _size;
    }

    void Attr()
    {
        struct shmid_ds ds;
        int n = shmctl(_shmid, IPC_STAT, &ds); // ds:输出型参数
        printf("shm_segsz: %ld\n", ds.shm_segsz);
        printf("key: 0x%x\n", ds.shm_perm.__key);
    }

    ~shm()
    {
        int n = shmdt(_startaddr);
        if (n == 0)
        {
            printf("shmdt success\n");
        }
        else
        {
            ERR_EXIT("shmdt");
        }
        if (_usertype == CREATER)
        {
            Destroy();
        }
    }

private:
    int _shmid;
    int _size;
    key_t _key;
    void *_startaddr;
    const std::string _usertype;
};